package cn.bouncyslime.algorithms.array.deRepeat;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 对指定整数数组指定重复次数的项进行去重，将其数量变为 1，在去重过程中，不改变各个项的原有顺序。
 * 假如某一项有 3 个，现在要对数量为 3 的项进行去重，则会去除该项的后两个。
 * <p>
 * 作者： 张一卓
 * 时间： 2019/12/20 11:19
 */
public class DeRepeat {
    /**
     * 对指定int数组中指定出现次数的项进行去重
     *
     * @param array  需要进行去重的数组
     * @param amount 出现次数，对出现次数为 amount 的项去重
     * @return int[] 去重之后的数组
     */
    public int[] resolve(int[] array, int amount) {
        // 判断数组是否为空
        if (array == null || array.length <= 0) {
            return null;
        }
        // 判断数组是否只有一个元素，则不需要进行处理
        if (array.length == 1 || amount <= 1) {
            return array;
        }
        // 用来统计数组中每一项出现的次数
        Map<Integer, Integer> amountMap = new HashMap<>();
        for (int num : array) {
            amountMap.put(num, amountMap.get(num) == null || amountMap.get(num) == 0 ? 1 : amountMap.get(num) + 1);
        }
        // 遍历 amountMap 找到出现数量为 amount 的元素，对其进行去重
        for (Map.Entry<Integer, Integer> entry : amountMap.entrySet()) {
            if (entry.getValue() == amount) {
                array = delete(array, entry.getKey(), amount);
            }
        }
        System.out.println(Arrays.toString(array));
        return array;
    }

    /**
     * 对指定int数组中值为deleteNum的项进行去重
     *
     * @param array     需要进行去重的数组
     * @param deleteNum 需要删除的数
     * @param amount    需要删除的数的出现次数
     * @return 对指定元素去重之后的数组
     */
    public int[] delete(int[] array, int deleteNum, int amount) {
        int[] tempArray = new int[array.length - amount + 1];
        int flag = 0;
        for (int i = 0; i < array.length; i++) {
            // 当该项不等于要去重的项的值时
            if (array[i] != deleteNum) {
                tempArray[flag] = array[i];
                flag++;
            } else if (array[i] == deleteNum && amount != 1) {   //当该项等于要去重的项的值，并且还没有遍历过该元素时
                tempArray[flag] = array[i];
                amount = 1;
                flag++;
            }
        }
        return tempArray;
    }
}